package tree.迭代bfs;

import po.TreeNode;

import java.util.*;

/**
 * @author songZiHao
 * @version 1.0.0
 * @ClassName 二叉树的层序遍历102.java
 * @Description TODO https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
 * @createTime 2021年03月26日 11:42:00
 */
public class 二叉树的层序遍历102 {


    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue=new LinkedList<>();
        List<List<Integer>> lists=new ArrayList<>();
        if (root==null){
            return lists;
        }
        queue.offer(root);
        while (!queue.isEmpty()){
            List<Integer> list=new ArrayList<>();
            //todo 循环Queue的长度次，表示的是上一层的节点的个数
            int size = queue.size();
            for (int i = 0; i <size ; i++) {
                TreeNode node = queue.poll();
                list.add(node.val);
                //todo 添加下一层
                if (node.left!=null){
                    queue.offer(node.left);
                }
                if (node.right!=null){
                    queue.offer(node.right);
                }
            }
            lists.add(list);
        }
        return lists;
    }
}
